import { invoke } from "@tauri-apps/api/core"
import { listen, UnlistenFn } from "@tauri-apps/api/event"

export interface UpdateHookEvent {
  operation: "INSERT" | "UPDATE" | "DELETE" | "UNKNOWN"
  database: string
  table: string
  rowid: number
}

export type UpdateHookCallback = (event: UpdateHookEvent) => void

let unlisten: UnlistenFn | null = null

export async function setupUpdateHook(
  dbPath: string,
  callback: UpdateHookCallback,
): Promise<void> {
  if (unlisten) {
    await removeUpdateHook(dbPath)
  }

  unlisten = await listen<UpdateHookEvent>("sqlite-update-hook", (event) => {
    callback(event.payload)
  })

  try {
    await invoke("plugin:sql|setup_update_hook", { db: dbPath })
  } catch (error) {
    console.error("Failed to setup update hook:", error)
    unlisten()
    unlisten = null
    throw error
  }
}

export async function removeUpdateHook(dbPath: string): Promise<void> {
  if (unlisten) {
    unlisten()
    unlisten = null
  }

  try {
    await invoke("plugin:sql|remove_update_hook", { db: dbPath })
  } catch (error) {
    console.error("Failed to remove update hook:", error)
    throw error
  }
}